from torchsummary import summary
from models.cv import VGG16, VGGBlock


def test_vgg_block():
    vgg_block = VGGBlock(3, 64, 2)
    summary(vgg_block, (3, 32, 32))


def test_vgg16():
    vgg16 = VGG16()
    summary(vgg16, (3, 32, 32))




if __name__ == '__main__':
    test_vgg16()


